package com.general.framework.mp;

import java.util.LinkedHashMap;
import java.util.Map;

public class LRUCache<K, V> extends LinkedHashMap<K, V> {
    private final int capacity;

    public LRUCache(int capacity) {
        // 调用父类的构造函数
        // 第一个参数是初始容量，第二个参数是加载因子，第三个参数表示是否按访问顺序排序
        super(capacity, 0.75f, true);
        this.capacity = capacity;
    }

    @Override
    protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
        // 当元素数量超过容量时，移除最久未使用的元素
        return size() > capacity;
    }

    public static void main(String[] args) {
        LRUCache<Integer, String> cache = new LRUCache<>(3);
        cache.put(1, "One");
        cache.put(2, "Two");
        cache.put(3, "Three");
        System.out.println(cache);

        // 访问元素 1，将其标记为最近使用
        cache.get(1);
        System.out.println(cache);

        // 添加新元素，会触发 LRU 淘汰策略，移除元素 2
        cache.put(4, "Four");
        System.out.println(cache);
    }
}